import torch
from torch import nn
from matplotlib import pyplot as plt

def __view_sigmod__():
    x = torch.linspace(-3,3,200)
    sigmodfunc = nn.Sigmoid()
    softmaxfunc = nn.Softmax()
    relufunc = nn.ReLU()
    tanhfunc = nn.Tanh()
    y1 = sigmodfunc(x)
    y2 = softmaxfunc(x)
    y3 = relufunc(x)
    y4 = tanhfunc(x)

    fig,ax = plt.subplots()
    ax.plot(x,y1)
    ax.plot(x,y2)
    ax.plot(x,y3)
    ax.plot(x,y4)
    plt.show()

if __name__ == "__main__":
    __view_sigmod__()